Pytorch Print Layers For%2